__author__ = 'ipetrov'
from internal_page import InternalPage

class MovieInfoPage(InternalPage):

    @property
    def remove_button(self):
        return self.driver.find_element_by_css_selector("img[alt=\"Remove\"]")

    @property
    def movie_title(self):
        return self.driver.find_element_by_class_name("maininfo_full").find_element_by_xpath("./h2")

    @property
    def is_this_page(self):
        return self.driver.find_element_by_css_selector("img[alt=\"Remove\"]")

    @property
    def alert(self):
        return self.driver.switch_to_alert()
